树 扁平化

/**
 * 增强版深度优先扁平化（保留层级和父级ID）
 * @param {Array} nodes - 树形数据
 * @param {number} level - 当前层级（默认0）
 * @param {string|number} parentId - 父节点ID（默认null）
 * @param {Array} result - 存储结果的数组
 * @returns {Array} - 包含层级和父ID的扁平数组
 */
function flattenTreeDFSWithParent(
  nodes,
  level = 0,
  parentId = null,
  result = []
) {
  nodes.forEach((node) => {
    const { children, ...rest } = node;
    result.push({
      ...rest, // 保留原属性
      level, // 当前层级
      parentId, // 父节点ID
    });
    if (children && children.length) {
      flattenTreeDFSWithParent(children, level + 1, node.id, result);
    }
  });
  return result;
}

const treeData = [
  {
    id: 1,
    name: "A",
    children: [
      { id: 2, name: "B", children: [{ id: 3, name: "C" }] },
      { id: 4, name: "D" },
    ],
  },
];

console.log(flattenTreeDFSWithParent(treeData));
// 输出结果示例：
// [
//   { id:1, name:'A', level:0, parentId:null },
//   { id:2, name:'B', level:1, parentId:1 },
//   { id:3, name:'C', level:2, parentId:2 },
//   { id:4, name:'D', level:1, parentId:1 }
// ]